-
Notifications
You must be signed in to change notification settings - Fork 142
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
paddle推理pipeline #381
base: develop
Are you sure you want to change the base?
paddle推理pipeline #381
Conversation
Thanks for your contribution! |
print("PaddleInfer pipeline saved to", output_path) | ||
|
||
|
||
# fd_pipe_cls = FastDeploySFastDeployStableDiffusionXLPipelineHousing |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这边是直接不用fd还是加个判断分别走呢?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
我重写一份代码,用paddle_infer_pipeline导出
parser = argparse.ArgumentParser() | ||
parser.add_argument( | ||
"--model_dir", | ||
default="runwayml/stable-diffusion-v1-5@fastdeploy", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个default是不是要改一下
@@ -444,6 +451,7 @@ def load_sub_model( | |||
try: | |||
# check if the module is in a subdirectory | |||
if os.path.isdir(os.path.join(cached_folder, name)): | |||
# import pdb; pdb.set_trace() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
无用的debug 注释记得删一下
|
支持绕过fastdeploy,静态图直接用paddle_inference推理